'''
simplist FCN
每个模型文件内必须有一个Model类
'''

import torch

class Model(torch.nn.Module):
    def __init__(self):
        """model fcy1, just for test
        """
        super(Model, self).__init__()
        print('initiating model fcy1')
        self.fc1 = torch.nn.Linear(12, 64)
        self.fc2 = torch.nn.Linear(64, 128)
        self.fc3 = torch.nn.Linear(128, 64)
        self.fc4 = torch.nn.Linear(64, 32)
        self.fc5 = torch.nn.Linear(32, 6)
        self.relu = torch.nn.ReLU()

    def forward(self, x):
        y = self.fc1(x)
        y = self.relu(y)
        y = self.fc2(y)
        y = self.relu(y)
        y = self.fc3(y)
        y = self.relu(y)
        y = self.fc4(y)
        y = self.relu(y)
        y = self.fc5(y)
        return y
